原始题目:剑指 Offer 57 - II. 和为s的连续正数序列 (opens new window)

解题思路:

求一段连续正数序列的和的公式,假设最小数为 aa,最大数为 bb,则这段数字的和为:

sum=(a+b)×(ba+1)2sum = \frac{(a+b) \times (b - a + 1)}{2}

使用双指针来解题,通过指针 llrr ,不断计算 [l,r][l, r] 窗口的序列和 sumsum,根据 sumsum 的大小调整窗口的大小。

  • 如果 sum=targetsum = target,说明这段序列和合法,可以记录起来,同时缩小窗口,ll 自增;
  • 如果 sum>targetsum > target,说明窗口太大,ll 自增;
  • 如果 sum<targetsum < target,说明窗口太小,rr 自增;

代码:

public int[][] findContinuousSequence(int target) {
    List<int[]> ans = new ArrayList<>();
    for (int l = 1, r = 2; l < r; ) {
        int sum = (l + r) * (r - l + 1) / 2;
        if (sum == target) {
            int[] list = new int[r - l + 1];
            for (int i = l; i <= r; i++) {
                list[i - l] = i;
            }
            ans.add(list);
            l++;
        } else if (sum < target) {
            r++;
        } else {
            l++;
        }
    }
    return ans.toArray(new int[ans.size()][]);
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

复杂度分析:

  • 时间复杂度O(N)O(N):其中 N=targetN = targetllrr 最终会相遇在 N2\frac{N}{2} 的位置。
  • 空间复杂度O(1)O(1):辅助变量占用常数大小的额外空间。
上次更新: 2023/10/15